DmlZipCreatorMojo.java

package pt.ist.fenixframework.dml.maven;

import java.io.File;
import java.io.FileOutputStream;
import java.net.MalformedURLException;
import java.net.URL;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.Comparator;
import java.util.List;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

import org.apache.commons.io.FileUtils;
import org.apache.maven.artifact.Artifact;
import org.apache.maven.artifact.DependencyResolutionRequiredException;
import org.apache.maven.model.Resource;
import org.apache.maven.plugin.AbstractMojo;
import org.apache.maven.plugin.MojoExecutionException;
import org.apache.maven.project.MavenProject;
import org.codehaus.plexus.util.DirectoryScanner;

import pt.ist.fenixframework.core.DmlFile;
import pt.ist.fenixframework.core.Project;

/**
 * Generate base main classes from the main DML files
 * 
 * @goal package-dmls
 * @phase generate-sources
 * @requiresDependencyResolution runtime
 */
public class DmlZipCreatorMojo extends AbstractMojo {

    /**
     * Maven Project
     * 
     * @parameter default-value="${project}"
     */
    protected MavenProject mavenProject;

    /**
     * File Source Directory
     * 
     * @parameter default-value="${basedir}/src/main/dml"
     * @readonly
     * @required
     */
    protected File dmlSourceDirectory;

    /**
     * Zip File Destination Directory
     * 
     * @parameter default-value="${project.build.directory}"
     * @readonly
     * @required
     */
    protected File zipDestinationDirectory;

    /**
     * Verbose Mode Flag
     * 
     * @parameter expression="${verbose}" default-value="false"
     */
    protected boolean verbose;

    @Override
    public void execute() throws MojoExecutionException {
        if (mavenProject.getArtifact().getType().equals("pom")) {
            getLog().info("Cannot compute dml files for pom projects");
            return;
        }

        Collection<String> compileClasspathElements;

        try {
            compileClasspathElements = mavenProject.getCompileClasspathElements();
        } catch (DependencyResolutionRequiredException e) {
            throw new MojoExecutionException(e.getMessage(), e);
        }

        List<String> classpath = new ArrayList<String>(compileClasspathElements);

        for (Artifact artifact : mavenProject.getDependencyArtifacts()) {
            if (artifact.getFile() != null && artifact.getFile().isDirectory()) {
                classpath.add(artifact.getFile().getAbsolutePath());
            }
        }

        DmlMojoUtils.augmentClassLoader(getLog(), classpath);

        List<URL> dmlFiles = new ArrayList<URL>();
        if (dmlSourceDirectory.exists()) {
            DirectoryScanner scanner = new DirectoryScanner();
            scanner.setBasedir(dmlSourceDirectory);

            String[] includes = { "**\\*.dml" };
            scanner.setIncludes(includes);
            scanner.scan();

            Resource resource = new Resource();
            resource.setDirectory(dmlSourceDirectory.getAbsolutePath());
            resource.addInclude("*.dml");
            mavenProject.addResource(resource);

            for (String includedFile : scanner.getIncludedFiles()) {
                String filePath = dmlSourceDirectory.getAbsolutePath() + "/" + includedFile;
                File file = new File(filePath);
                try {
                    dmlFiles.add(file.toURI().toURL());
                } catch (MalformedURLException e) {
                    getLog().error(e);
                }
            }
            Collections.sort(dmlFiles, new Comparator<URL>() {
                @Override
                public int compare(URL o1, URL o2) {
                    return o1.toExternalForm().compareTo(o2.toExternalForm());
                }
            });
        }

        try {
            Project project =
                    DmlMojoUtils.getProject(mavenProject, dmlSourceDirectory, zipDestinationDirectory, dmlFiles, getLog(),verbose);

            List<URL> dmls = new ArrayList<URL>();
            for (DmlFile dmlFile : project.getFullDmlSortedList()) {
                dmls.add(dmlFile.getUrl());
            }

            if (dmls.isEmpty()) {
                getLog().info("No dml files found");
                return;
            } else {
                File zipFile = new File(String.format("%s/%s_dmls.zip", zipDestinationDirectory, mavenProject.getArtifactId()));
                zipFile.getParentFile().mkdirs();
                ZipOutputStream out = new ZipOutputStream(new FileOutputStream(zipFile));
                Integer i = 1;
                for (URL dmlURL : dmls) {
                    String name = String.format("/dmls/domain_model_%02d.dml", i++);
                    ZipEntry e = new ZipEntry(name);
                    out.putNextEntry(e);
                    File f = new File(zipDestinationDirectory + name);
                    FileUtils.copyURLToFile(dmlURL, f);
                    out.write(FileUtils.readFileToByteArray(f));
                    out.closeEntry();
                }
                out.close();
            }

        } catch (Exception e) {
            getLog().error(e);
        }
    }
}